package com.lhkj.ct.framework.mybatis.handlers;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.core.toolkit.Constants;
import com.lhkj.ct.base.enums.DeleteStatus;
import com.lhkj.ct.base.utils.ShiroUtil;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.util.Date;
import java.util.Objects;
import java.util.Optional;

/**
 * TODO 元对象处理程序
 */
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

    /**
     * 创建人id
     */
    private static final String CREATE_BY = "createBy";

    /**
     * 创建时间
     */
    private static final String CREATE_TIME = "createTime";

    /**
     * 修改人iid
     */
    private static final String UPDATE_BY = "updateBy";

    /**
     * 修改时间
     */
    private static final String UPDATE_TIME = "updateTime";

    /**
     * 删除状态
     */
    private static final String DEL_FLAG = "delFlag";

    /**
     * 版本号
     */
    private static final String VERSION = "version";

    /**
     * 配置插入时需要填充的字段/注意该写法是3.3.2版本之后的 tips 只支持 int,Integer,long,Long,Date,Timestamp,LocalDateTime, Enum这些数据类型
     *
     * @param metaObject 数据库元对象
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        // 插入数据填充插入时间
        this.strictInsertFill(metaObject, CREATE_TIME, Date.class, new Date());
        this.strictInsertFill(metaObject, CREATE_TIME, LocalDateTime.class, LocalDateTime.now());
        // 插入数据填充默认版本号为1
        this.strictInsertFill(metaObject, VERSION, Integer.class, 1);
        // 插入数据填充默认删除状态为0
        this.strictInsertFill(metaObject, DEL_FLAG, DeleteStatus.class, DeleteStatus.NORMAL);
        // 插入数据填充默认创建人
        if (isFills(CREATE_BY, metaObject, FieldFill.INSERT)) {
            this.strictUpdateFill(metaObject, CREATE_BY, String.class, ShiroUtil.getAuthUserInfo().getUserId());
        }
    }

    /**
     * 配置更新时需要填充的字段
     *
     * @param metaObject 元对象
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        // 更新数据填充更新时间
        this.strictUpdateFill(metaObject, UPDATE_TIME, Date.class, new Date());

        this.strictInsertFill(metaObject, UPDATE_TIME, LocalDateTime.class, LocalDateTime.now());
        // 插入数据填充默认创建人
        if (isFills(UPDATE_BY, metaObject, FieldFill.UPDATE)) {
            this.strictUpdateFill(metaObject, UPDATE_BY, String.class, ShiroUtil.getAuthUserInfo().getUserId());
        }
    }

    /**
     * @Description 重构父级该方法 校验是否存在默认值注解 并且对创建人跟修改人赋值 创建时间在一开始就赋值了
     * @param fieldName 字段名
     * @param metaObject mybatis对应的entity注解反射类
     * @param fieldFill entity注解
     * @return
     */
    public boolean isFills(String fieldName, MetaObject metaObject, FieldFill fieldFill) {
        TableInfo tableInfo = metaObject.hasGetter(Constants.MP_OPTLOCK_VERSION_ORIGINAL) ?
                TableInfoHelper.getTableInfo(metaObject.getValue(Constants.MP_OPTLOCK_VERSION_ORIGINAL).getClass())
                : TableInfoHelper.getTableInfo(metaObject.getOriginalObject().getClass());
        if (Objects.nonNull(tableInfo)) {
            Optional<TableFieldInfo> first = tableInfo.getFieldList().stream()
                    //v_3.1.1+ 设置子类的值也可以通过
                    .filter(e -> e.getProperty().equals(fieldName))
                    .findFirst();
            return first.isPresent() && Objects.equals(first.get().getFieldFill(), fieldFill);
        }
        return false;
    }

    /**
     * @Description 重构父级该方法对应字段进行赋值
     * @param fieldName
     * @param fieldVal
     * @param metaObject
     * @param fieldFill
     * @return
     */
    public MetaObjectHandler setFieldValByNames(String fieldName, Object fieldVal, MetaObject metaObject, FieldFill fieldFill) {
        // FieldFill.INSERT
        if (metaObject.hasSetter(fieldName) && metaObject.hasGetter(fieldName) && isFills(fieldName, metaObject, fieldFill)) {
            if (CREATE_BY.equals(fieldName) || UPDATE_BY.equals(fieldName)) {
                metaObject.setValue(fieldName, fieldVal.toString());
            } else {
                metaObject.setValue(fieldName, fieldVal);
            }
        } else if (metaObject.hasGetter(Constants.ENTITY)) {
            // FieldFill.INSERT_UPDATE
            // 此处是进行修改操作给修改人跟修改时间进行赋值的时候 MetaObject对象的originalObject属性跟保存是结构不一样 具体自己大断点看
            Object et = metaObject.getValue(Constants.ENTITY);
            if (et != null) {
                MetaObject etMeta = SystemMetaObject.forObject(et);
                if (etMeta.hasSetter(fieldName) && isFills(fieldName,  etMeta, fieldFill)) {
                    if(ObjectUtils.isEmpty(fieldVal)){
                        etMeta.setValue(fieldName, null);
                    }else if ((CREATE_BY.equals(fieldName) || UPDATE_BY.equals(fieldName))) {
                        etMeta.setValue(fieldName, fieldVal.toString());
                    }else{
                        etMeta.setValue(fieldName, fieldVal);
                    }
                }
            }
        }
        return this;
    }



}
